// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_BROADCAST_CHANNEL_BROADCAST_CHANNEL_PROVIDER_H_
#define CONTENT_BROWSER_BROADCAST_CHANNEL_BROADCAST_CHANNEL_PROVIDER_H_

#include <map>

#include "base/memory/ref_counted.h"
#include "mojo/public/cpp/bindings/binding_set.h"
#include "third_party/WebKit/public/platform/modules/broadcastchannel/broadcast_channel.mojom.h"
#include "url/origin.h"

namespace content {

class BroadcastChannelProvider
    : public base::RefCountedThreadSafe<BroadcastChannelProvider>,
      public blink::mojom::BroadcastChannelProvider {
public:
    BroadcastChannelProvider();
    void Connect(
        mojo::InterfaceRequest<blink::mojom::BroadcastChannelProvider> request);

    void ConnectToChannel(
        const url::Origin& origin,
        const std::string& name,
        blink::mojom::BroadcastChannelClientAssociatedPtrInfo client,
        blink::mojom::BroadcastChannelClientAssociatedRequest connection)
        override;

private:
    friend class base::RefCountedThreadSafe<BroadcastChannelProvider>;
    class Connection;

    ~BroadcastChannelProvider() override;

    void UnregisterConnection(Connection*);
    void ReceivedMessageOnConnection(Connection*,
        const std::vector<uint8_t>& message);

    mojo::BindingSet<blink::mojom::BroadcastChannelProvider> bindings_;
    std::map<url::Origin, std::multimap<std::string, std::unique_ptr<Connection>>>
        connections_;
};

} // namespace content

#endif // CONTENT_BROWSER_BROADCAST_CHANNEL_BROADCAST_CHANNEL_PROVIDER_H_
